import sys
import m3u8dl
import argparse
import os

def downloadCallback(mp4path) :
	print('下载完成:')
	print(mp4path)

if __name__ == "__main__":  
    parser = argparse.ArgumentParser(description='Download M3U8 files.')  
    parser.add_argument('m3u8url', type=str, help='The URL of the M3U8 file.')  
    parser.add_argument('filename', type=str, help='The name of the output file.')  
    args = parser.parse_args()  
  
    md = m3u8dl.M3u8Downloader(downloadCallback)
    result = md.downloadM3u8(args.m3u8url, args.filename, './m3u8dlDownloads')
    # ffmpegcmd = md.getFFmpegCommand()
    # print(f"Download result: {ffmpegcmd}")  # 假设downloadM3u8返回一个表示结果的值
    # print(result)
    if result == 1 :
        print("下载完成")
        with open(f'./m3u8dlDownloads/{args.filename}/ret.data', 'w') as file:  
            # 将数字 1 转换为字符串，并写入文件  
            # 在 Python 中，write 方法接受一个字符串作为参数  
            file.write(str(1))  
            print("写入ret.data文件成功")